package io.github.talelin.latticy.common.util;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.WriterException;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.oned.Code128Writer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import io.github.talelin.core.util.DateUtil;
import org.apache.commons.lang3.StringUtils;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;


/**
 * @ClassName：GoogleBarCodeUtils
 * @Author: ljj
 * @Date: 2022/5/24 17:40
 */
public class GoogleBarCodeUtils {
    /**
     * 条形码宽度
     */
    private static final int WIDTH = 100;

    /**
     * 条形码高度
     */
    private static final int HEIGHT = 30;

    /**
     * 加文字 条形码
     */
    private static final int WORD_HEIGHT = 110;

    /**
     * 条形码右下角第一段数据
     */
    private static final String RIGHT_FIRST_WORDS = "superMan";

    private static final String RIGHT_SECOND_WORDS = "Made in ";

    private static final String CHINA = "China";

    /**
     * 设置 条形码参数
     */
    private static Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>() {
        private static final long serialVersionUID = 1L;

        {
            // 设置编码方式
            put(EncodeHintType.CHARACTER_SET, "utf-8");
            // 容错级别 这里选择最高H级别
            put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
            put(EncodeHintType.MARGIN, 1);
        }
    };

    /**
     * 生成 图片缓冲
     *
     * @param vaNumber VA 码
     * @return 返回BufferedImage
     * @author fxbin
     */
    public static BufferedImage getBarCode(String vaNumber, int width, int heght) {
        try {
            Code128Writer writer = new Code128Writer();
            // 编码内容, 编码类型, 宽度, 高度, 设置参数
            BitMatrix bitMatrix = writer.encode(vaNumber, BarcodeFormat.CODE_128, width, heght, null);
            return MatrixToImageWriter.toBufferedImage(bitMatrix);
        } catch (WriterException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 生成 图片缓冲
     *
     * @param vaNumber VA 码
     * @return 返回BufferedImage
     * @author fxbin
     */
    public static BufferedImage getBarCode(String vaNumber) {
        try {
            Code128Writer writer = new Code128Writer();
            // 编码内容, 编码类型, 宽度, 高度, 设置参数
            BitMatrix bitMatrix = writer.encode(vaNumber, BarcodeFormat.CODE_128, WIDTH, HEIGHT, hints);
            return MatrixToImageWriter.toBufferedImage(bitMatrix);
        } catch (WriterException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 把带logo的二维码下面加上文字
     *
     * @param image 条形码图片
     * @param words 文字
     * @return 返回BufferedImage
     * @author fxbin
     */
    public static BufferedImage insertWords(BufferedImage image, String code, String name, String enName,
                                            String date, String origin, int width, int heght) {

        // 新的图片，把带logo的二维码下面加上文字
        if (StringUtils.isNotEmpty(code)) {
            BufferedImage outImage = new BufferedImage(width, heght + 30, BufferedImage.TYPE_INT_RGB);
            Graphics2D g2d = outImage.createGraphics();
            // 抗锯齿
            setGraphics2D(g2d);
            // 设置白色
            setColorWhite(g2d);
            // 画条形码到新的面板
            g2d.drawImage(image, 0, 0, image.getWidth(), image.getHeight(), null);
            // 画文字到新的面板
            Color color = new Color(0, 0, 0);
            g2d.setColor(color);
            // 字体、字型、字号
            g2d.setFont(new Font("微软雅黑", Font.PLAIN, 12));
            int width2 = outImage.getWidth() - image.getWidth();
            int heght2 = outImage.getHeight() - image.getHeight();
            int strHeight = 10;
            drawString(code, width2, heght2, g2d, strHeight);
            if (StringUtils.isNotEmpty(name)) {
                strHeight += 10;
                drawString(name, width2, heght2, g2d, strHeight);
            }
            if (StringUtils.isNotEmpty(enName)) {
                strHeight += 10;
                drawString(enName, width2, heght2, g2d, strHeight);
            }
            if (StringUtils.isNotEmpty(date)) {
                strHeight += 10;
                drawString(date, width2, heght2, g2d, strHeight);
            }
            if (StringUtils.isNotEmpty(origin)) {
                strHeight += 10;
                drawString(RIGHT_SECOND_WORDS + origin, width2, heght2, g2d, strHeight);
            }
            g2d.dispose();
            outImage.flush();
            return outImage;
        }
        return null;
    }

    private static void drawString(String str, int width, int heght, Graphics2D g2d, int strHeght) {
        int strWidth = g2d.getFontMetrics().stringWidth(str);
        int startX = (width - strWidth) / 2;
        int startY = heght + strHeght;
        g2d.drawString(str, startX, startY);
    }


    /**
     * 把带logo的二维码下面加上文字
     *
     * @param image 条形码图片
     * @param words 文字
     * @return 返回BufferedImage
     * @author fxbin
     */
    public static BufferedImage insertWords(BufferedImage image, String words, String name,
                                            String time, String origin) {
        // 新的图片，把带logo的二维码下面加上文字
        if (StringUtils.isNotEmpty(words)) {
            BufferedImage outImage = new BufferedImage(WIDTH, WORD_HEIGHT, BufferedImage.TYPE_INT_RGB);
            Graphics2D g2d = outImage.createGraphics();
            // 抗锯齿
            setGraphics2D(g2d);
            // 设置白色
            setColorWhite(g2d);
            // 画条形码到新的面板
            g2d.drawImage(image, 0, 0, image.getWidth(), image.getHeight(), null);
            // 画文字到新的面板
            Color color = new Color(0, 0, 0);
            g2d.setColor(color);
            // 字体、字型、字号
            g2d.setFont(new Font("微软雅黑", Font.PLAIN, 9));
            //文字长度
            int strWidth = g2d.getFontMetrics().stringWidth(words);
            //总长度减去文字长度的一半  （居中显示）
            int wordStartX = (WIDTH - strWidth) / 2;
            //height + (outImage.getHeight() - height) / 2 + 12
            int wordStartY = HEIGHT + 8;
            // time 文字长度
            int timeWidth = 1;
            // rightFirstWords 文字长度
            int rightFirstWordsWidth = WIDTH - g2d.getFontMetrics().stringWidth(name);
            // rightSecondWords 文字长度
            String str = StringUtils.isEmpty(origin) ? CHINA : origin;
            int rightSecondWordsWidth = WIDTH - g2d.getFontMetrics().stringWidth(RIGHT_SECOND_WORDS + str);
            // 画文字
            g2d.drawString(words, wordStartX, wordStartY);
            // g2d.drawString(time, timeWidth, wordStartY + 18);
            g2d.drawString(name, rightFirstWordsWidth, wordStartY + 8);
            g2d.drawString(RIGHT_SECOND_WORDS + str, rightSecondWordsWidth, wordStartY + 18);
            g2d.dispose();
            outImage.flush();
            return outImage;
        }
        return null;
    }

    /**
     * 设置 Graphics2D 属性  （抗锯齿）
     *
     * @param g2d Graphics2D提供对几何形状、坐标转换、颜色管理和文本布局更为复杂的控制
     */
    private static void setGraphics2D(Graphics2D g2d) {
        // 消除画图锯齿
        g2d.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
        // 消除文字锯齿
        g2d.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
        g2d.setRenderingHint(RenderingHints.KEY_STROKE_CONTROL, RenderingHints.VALUE_STROKE_DEFAULT);
        Stroke s = new BasicStroke(1, BasicStroke.CAP_ROUND, BasicStroke.JOIN_MITER);
        g2d.setStroke(s);
    }

    /**
     * 设置背景为白色
     *
     * @param g2d Graphics2D提供对几何形状、坐标转换、颜色管理和文本布局更为复杂的控制
     */
    private static void setColorWhite(Graphics2D g2d) {
        g2d.setColor(Color.WHITE);
        //填充整个屏幕
        g2d.fillRect(0, 0, 302, 113);
        //设置笔刷
        g2d.setColor(Color.BLACK);
    }

    // public static void main(String[] args) throws IOException {
    //     BufferedImage image = insertWords(getBarCode("sku-01"), "sku-01", DateUtils.dateStr2(new Date()), "China");
    //     ImageIO.write(image, "jpg", new File("D:/superMan.png"));
    // }
}
